Array Expansion

This is an example using the Omtools expand operation in order to expand an array to a higher order array.

from csdl_om import Simulatorimport csdlfrom csdl import Modelimport numpy as np

class ExampleArray2HigherArray(Model):
    def define(self):        # Expanding an array into a higher-rank array        val = np.array([            [1., 2., 3.],            [4., 5., 6.],        ])        array = self.declare_variable('array', val=val)        expanded_array = csdl.expand(array, (2, 4, 3, 1), 'ij->iajb')        self.register_output('expanded_array', expanded_array)

sim = Simulator(ExampleArray2HigherArray())sim.run()
print('array', sim['array'].shape)print(sim['array'])print('expanded_array', sim['expanded_array'].shape)print(sim['expanded_array'])

[[1. 2. 3.] [4. 5. 6.]]expanded_array (2, 4, 3, 1)[[[[1.]   [2.]   [3.]]
  [[1.]   [2.]   [3.]]
  [[1.]   [2.]   [3.]]
  [[1.]   [2.]   [3.]]]

 [[[4.]   [5.]   [6.]]
  [[4.]   [5.]   [6.]]
  [[4.]   [5.]   [6.]]
  [[4.]   [5.]   [6.]]]]